﻿#ifndef _CELLTHREAD_HPP_
#define _CELLTHREAD_HPP_

#include "CELLSemaphore.hpp"
#include <functional>
#include <thread>
#include <mutex>

class CELLThread
{
private:
	typedef std::function<void(CELLThread*)> EventCall;
public:
	// 睡眠,单位微秒
	static void Sleep(int i)
	{
		std::chrono::milliseconds t(i);
		std::this_thread::sleep_for(t);
	}

	// 启动本线程，要注册线程事件直接传函数指针进来
	void Start(EventCall onCreate = nullptr, 
		EventCall onRun = nullptr, 
		EventCall onDestory = nullptr)
	{
		std::lock_guard<std::mutex> lg(_mutex);
		if (!_isRun)
		{
			_isRun = true;
			if (onCreate)
				_onCreate = onCreate;
			if (onRun)
				_onRun = onRun;
			if (onDestory)
				_onDestory = onDestory;
			// 将线程启动放在后面，确保传进线程的本对象的三个事件都被准确赋值
			std::thread t(&CELLThread::OnWork, this);
			t.detach();
		}
	}

	// 关闭线程
	void Close()
	{
		std::lock_guard<std::mutex> lg(_mutex);
		if (_isRun)
		{
			_isRun = false;
			_sem.wait();
		}
	}

	// 在工作函数中退出
	// 不需要使用信号量来阻塞等待
	// 如果使用会阻塞
	void Exit()
	{
		std::lock_guard<std::mutex> lg(_mutex);
		if (_isRun)
		{
			_isRun = false;
		}
	}

	// 线程是否处于运行状态
	bool isRun()
	{
		return _isRun;
	}
protected:
	// 线程入口函数(线程工作函数)
	void OnWork()
	{
		// 如果注册了线程启动事件，就执行
		if (_onCreate)
			_onCreate(this);
		if (_onRun)
			_onRun(this);
		if (_onDestory)
			_onDestory(this);

		_sem.wakeup();
		_isRun = false;
	}

private:
	// 创建时
	EventCall _onCreate;
	// 运行时
	EventCall _onRun;
	// 退出时
	EventCall _onDestory;
	// 控制线程的终止、退出
	CELLSemaphore _sem;
	// 互斥量
	std::mutex _mutex;
	// 线程是否运行中
	bool _isRun = false;
};

#endif // #ifndef _CELLTHREAD_HPP_
